package com.supplychain.scm.core.common.validator.range;

import cn.hutool.core.convert.Convert;
import cn.hutool.core.util.ArrayUtil;
import cn.hutool.core.util.StrUtil;

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

public class InEnumValidator implements ConstraintValidator<InEnum, Object> {

    private List<String> values;

    @Override
    public void initialize(InEnum annotation) {
        //先读取可选项
        if (ArrayUtil.isNotEmpty(annotation.options())) {
            values = Arrays.asList(annotation.options());
        } else {
            ListValuable[] enumConstants = annotation.value().getEnumConstants();
            if (enumConstants != null) {
                if (enumConstants.length == 0) {
                    this.values = Collections.emptyList();
                } else {
                    this.values = enumConstants[0].listValues();
                }
            } else {
                this.values = Collections.emptyList();
            }
        }
    }

    @Override
    public boolean isValid(Object value, ConstraintValidatorContext context) {
        // 为空时，默认不校验，即认为通过
        if (value == null) {
            return true;
        }
        // 校验通过
        if (values.contains(Convert.toStr(value))) {
            return true;
        }
        // 校验不通过，自定义提示语句（因为，注解上的 value 是枚举类，无法获得枚举类的实际值）
        context.disableDefaultConstraintViolation(); // 禁用默认的 message 的值
        context.buildConstraintViolationWithTemplate(StrUtil.format(context.getDefaultConstraintMessageTemplate(), values.toString())).addConstraintViolation(); // 重新添加错误提示语句
        return false;
    }

}

